import pandas as pd
from sklearn.datasets import load_iris

# 加载鸢尾花数据集 -- 创建一个二分类问题 测试逻辑回归通用代码
iris = load_iris()
data = iris.data
target = iris.target

# 简化为二分类问题：只保留山鸢尾（0）和维吉尼亚鸢尾（1），丢弃第三类（Virginica）
binary_target = (target != 2).astype(int)  # 将非Virginica类别设为1，Virginica类别设为0

# 创建DataFrame以方便查看
df = pd.DataFrame(data, columns=iris.feature_names)
df['species_binary'] = target

# 将DataFrame保存为xlsx文件
df.to_excel('01_train.xlsx', index=False)

